// Copyright 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "content/renderer/web_ui_extension_data.h"

#include "content/common/view_messages.h"
#include "content/public/renderer/render_view.h"

namespace content {

WebUIExtensionData::WebUIExtensionData(RenderView* render_view)
    : RenderViewObserver(render_view)
    , RenderViewObserverTracker<WebUIExtensionData>(render_view)
{
}

WebUIExtensionData::~WebUIExtensionData()
{
}

std::string WebUIExtensionData::GetValue(const std::string& key) const
{
    std::map<std::string, std::string>::const_iterator it = variable_map_.find(key);
    if (it == variable_map_.end())
        return std::string();
    return it->second;
}

bool WebUIExtensionData::OnMessageReceived(const IPC::Message& message)
{
    bool handled = true;
    IPC_BEGIN_MESSAGE_MAP(WebUIExtensionData, message)
    IPC_MESSAGE_HANDLER(ViewMsg_SetWebUIProperty, OnSetWebUIProperty)
    IPC_MESSAGE_UNHANDLED(handled = false)
    IPC_END_MESSAGE_MAP()
    return handled;
}

void WebUIExtensionData::OnSetWebUIProperty(const std::string& name,
    const std::string& value)
{
    variable_map_[name] = value;
}

void WebUIExtensionData::OnDestruct()
{
    delete this;
}

} // namespace content
